package company.string;

import java.util.*;

/**
 * 49. 字母异位词分组
 * 给你一个字符串数组，请你将 字母异位词 组合在一起。可以按任意顺序返回结果列表。
 * <p>
 * 字母异位词 是由重新排列源单词的所有字母得到的一个新单词。
 * <p>
 *  
 * <p>
 * 示例 1:
 * <p>
 * 输入: strs = ["eat", "tea", "tan", "ate", "nat", "bat"]
 * 输出: [["bat"],["nat","tan"],["ate","eat","tea"]]
 * 示例 2:
 * <p>
 * 输入: strs = [""]
 * 输出: [[""]]
 * 示例 3:
 * <p>
 * 输入: strs = ["a"]
 * 输出: [["a"]]
 * <p>
 * 来源：力扣（LeetCode）
 * 链接：https://leetcode.cn/problems/group-anagrams
 * 著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。
 */
public class GroupAnagrams49 {
    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        String input = sc.nextLine();
        String[] split = input.replace("[", "").
                replaceAll("]", "").
                replace("\"", "")
                .split(",",-1);
        List<List<String>> lists = groupAnagrams(split);
        for (List<String> list : lists) {
            System.out.println(list);
        }
    }

    //
    public static List<List<String>> groupAnagrams(String[] strs) {
        List<String> strList = new ArrayList<>();
        strList.addAll(Arrays.asList(strs));
        List<List<String>> allList = new ArrayList<>();
        List<String> nullStr = new ArrayList<>();
        // 空格处理
        for (String s : strList) {
            if (s.length()==0) {
                nullStr.add(s);
            }
        }
        if (nullStr.size()>0) {
            allList.add(nullStr);
            strList.removeAll(nullStr);
        }

        while (strList.size() > 0) {
            String str = strList.get(0);
            TreeSet<String> subList = new TreeSet<>();
            char[] chars = str.toCharArray();
            // 求组合
            dfs(chars, 0, subList);
            // 求交集
            List<String> copy = new ArrayList<>(strList);
            copy.retainAll(subList);
            allList.add(new ArrayList<>(copy));
            // 移除已经匹配的字典
            strList.removeAll(copy);
        }

        // 排序
        return allList;
    }

    // 全排列
    private static void dfs(char[] word, int index, TreeSet<String> subList) {
        if (index == word.length) {
            subList.add(String.valueOf(word));
            return;
        }

        for (int i = index; i < word.length; i++) {
            // 遍历字符，依次和第二、第三等依次交换位置
            combinate(word, index, i);
            // 开始以第二个字符为主，进行做交换
            dfs(word, index + 1, subList);
            combinate(word, index, i);
        }
    }

    private static void combinate(char[] chars, int begin, int end) {
        char tmp = chars[begin];
        chars[begin] = chars[end];
        chars[end] = tmp;
    }

    // 官方
    public List<List<String>> groupAnagramsTwo(String[] strs) {
        Map<String, List<String>> map = new HashMap<String, List<String>>();
        for (String str : strs) {
            char[] array = str.toCharArray();
            Arrays.sort(array);
            String key = new String(array);
            List<String> list = map.getOrDefault(key, new ArrayList<String>());
            list.add(str);
            map.put(key, list);
        }
        return new ArrayList<List<String>>(map.values());
    }
}
